import pygame.sprite

try:
    from plane_sprites import *
except ModuleNotFoundError:
    print('没有pygame依赖包,在 cmd 中执行 \'pip install pygame \'')

FPS = 60
CREATE_ENEMY = pygame.USEREVENT
HERO_FIRE = pygame.USEREVENT + 1


# 定义飞机大战的类
class GamePlane:
    def __init__(self):
        # 创建游戏窗口
        self.screen = pygame.display.set_mode(SCREEN_RECT.size)
        # 更改标题
        pygame.display.set_caption('飞机大战')
        # 创建游戏时钟
        self.clock = pygame.time.Clock()
        self.__create_sprites()
        pygame.time.set_timer(CREATE_ENEMY, 1000)
        pygame.time.set_timer(HERO_FIRE, 300)
        # 开始游戏
        self.start_game()

    # 创建精灵/精灵组
    def __create_sprites(self):
        bg1 = Background()
        bg2 = Background(True)
        self.bg_group = pygame.sprite.Group(bg1, bg2)
        self.enemy_group = pygame.sprite.Group()
        self.hero = Hero()
        self.hero_group = pygame.sprite.Group(self.hero)

    def start_game(self):
        print('开始游戏....')
        while True:
            # 设置帧率
            self.clock.tick(FPS)
            # 事件监听
            self.__event_handler()
            # 碰撞检测
            self.__check_collide()
            # 更新/绘制精灵组
            self.__update_sprites()
            # 更新显示
            pygame.display.update()

    def __event_handler(self):
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                GamePlane.game_over()
            elif event.type == CREATE_ENEMY:
                print('敌机入场...')
                enemy = Enemy()
                self.enemy_group.add(enemy)
            elif event.type == HERO_FIRE:

                self.hero.fire()

        # 按键元组
        keys_pressed = pygame.key.get_pressed()
        if keys_pressed[pygame.K_RIGHT]:
            self.hero.speed = 2
        elif keys_pressed[pygame.K_LEFT]:
            self.hero.speed = -2
        else:
            self.hero.speed = 0

    def __check_collide(self):
        # 1>子弹摧毁敌机
        # 参数1:一个精灵组
        # 参数2:一个精灵组
        # 参数3:True:当第一个精灵组中的精灵碰撞到了第二个精灵组中的精灵,那么第一个精灵组中的精灵就会被销毁 False:不会销毁
        # 参数4:True:当第二个精灵组中的精灵碰撞到了第一个精灵组中的精灵,那么第二个精灵组中的精灵就会被销毁 False:不会销毁
        pygame.sprite.groupcollide(self.hero.bullet_group, self.enemy_group, True, True)
        # 2>敌机摧毁英雄
            # 参数1:精灵
            # 参数2:精灵组
            # 参数3: True:当精灵碰撞到了精灵组中的精灵 或者 精灵组中的精灵碰撞到了精灵,那么精灵组中的精灵就会被销毁  False: 不会销毁
            # return :返回一个数组,数组中存放的是被销毁的精灵
        enemies =  pygame.sprite.spritecollide(self.hero, self.enemy_group, True)
        if len(enemies) > 0:
            self.hero.kill()
            GamePlane.game_over()

    def __update_sprites(self):
        self.bg_group.update()
        self.bg_group.draw(self.screen)
        self.enemy_group.update()
        self.enemy_group.draw(self.screen)
        self.hero_group.update()
        self.hero_group.draw(self.screen)
        self.hero.bullet_group.update()
        self.hero.bullet_group.draw(self.screen)

    @staticmethod
    def game_over():
        print('游戏结束...')
        pygame.quit()
        exit()


if __name__ == '__main__':
    GamePlane()
